package com.spzx.product.service.impl;

import com.spzx.product.domain.Category;
import com.spzx.product.mapper.CategoryMapper;
import com.spzx.product.service.ICategoryService;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;

import java.util.*;

import com.spzx.product.vo.CategoryVo;
import org.springframework.beans.BeanUtils;
import org.springframework.stereotype.Service;
import org.springframework.util.CollectionUtils;

/**
 * <p>
 * 商品分类 服务实现类
 * </p>
 *
 * @author gax
 * @since 2025-08-23
 */
@Service
public class CategoryServiceImpl extends ServiceImpl<CategoryMapper, Category> implements ICategoryService {
    @Override
    public List<Category> treeSelect(Long parentId) {
        // 根据传入父级分类 ID 查询分类列表
        LambdaQueryWrapper<Category> lambdaQueryWrapper = new LambdaQueryWrapper<>();
        lambdaQueryWrapper.eq(Category::getParentId, parentId)
                .select(Category::getId, Category::getName, Category::getImageUrl, Category::getParentId,
                        Category::getStatus, Category::getOrderNum, Category::getCreateTime);
        // 调用 MybatisPlus 提供的 list 方法，查询分类列表
        List<Category> categoryList = this.list(lambdaQueryWrapper);

        if (!CollectionUtils.isEmpty(categoryList)) {
            // 判断分类列表中每个分类是否还有子类，如果有，将 hashChildren 设置为 true
            categoryList.forEach(elm -> {
                LambdaQueryWrapper<Category> queryWrapper = new LambdaQueryWrapper<Category>().eq(Category::getParentId,
                        elm.getId());
                long count = this.count(queryWrapper);
                elm.setHasChildren(count > 0);
            });
        }
        return categoryList;
    }

    @Override
    public List<CategoryVo> getLevelOneCategory() {
        // 一级分类：parentId 为 0 的是一级分类
        List<Category> categoryList = this.list(new LambdaQueryWrapper<Category>().eq(Category::getParentId, 0));
        // 封装 vo
        if (!CollectionUtils.isEmpty(categoryList)) {
            List<CategoryVo> categoryVoList = categoryList.stream().map(elm -> {
                CategoryVo categoryVo = new CategoryVo();
                // 使用 BeanUtils 工具类的前提是：目标对象和源对象的属性名称和类型相同
                BeanUtils.copyProperties(elm, categoryVo);
                return categoryVo;
            }).toList();
            return categoryVoList;
        }
        return null;
    }

    @Override
    public List<CategoryVo> tree() {
        // ① 查询所有的分类目录
        List<Category> allCategoryList = this.list();
        if (CollectionUtils.isEmpty(allCategoryList)) {
            // 如果分类目录列表为空，返回空列表
            return Collections.emptyList();
        }
        // 转换为 VO，并初始化 children
        List<CategoryVo> categoryVoList = allCategoryList.stream().map(item -> {
            CategoryVo vo = new CategoryVo();
            BeanUtils.copyProperties(item, vo);
            return vo;
        }).toList();

        // ② 建立 id -> node 索引
        // 具体来说，使用 Map 结构存储分类 ID 和对应的分类目录对象，方便后续快速查找
        // Map 的 key 是分类 ID，value 是目录对象
        Map<Long, CategoryVo> idMap = new HashMap<>(categoryVoList.size() * 2);
        for (CategoryVo vo : categoryVoList) {
            idMap.put(vo.getId(), vo);
        }

        // ③ 一次遍历，组装父子关系
        // 用 roots 集合存储一级目录，即 parentID 为 0 的目录，这些是顶级目录
        // 思路是：只要把 roots 集合中的这些顶级目录的 children 属性设置为对应的子目录，就完成了分类层级树
        List<CategoryVo> roots = new ArrayList<>();
        for (CategoryVo vo : categoryVoList) {

            // 首先，获取当前目录的父目录 ID
            Long parentId = vo.getParentId();
            
            if (parentId == null || parentId == 0L) {
                // 如果父目录 ID 是 0，则表示当前目录是一级目录，直接添加到 roots 集合
                roots.add(vo);
            } else {
                // 如果当前目录不是一级目录，则根据父目录 ID，从 Map 中获取父目录对象
                // 然后利用父目录的对象，向其孩子属性中，添加当前遍历的目录对象即可
                CategoryVo parent = idMap.get(parentId);
                if (parent != null) {
                    parent.getChildren().add(vo);
                } else {
                    // 如果找不到父亲，可选：丢到根节点列表（视业务需求）
                    roots.add(vo);
                }
            }
        }
        return roots;
    }
}
